#!/bin/bash
# 说明：
# 本脚本执行以下几个步骤： 软件安装、CPU压测、GPU压测、监控开启
# CPU压测提供CPU核数选择，默认是4核，请根据需求修改
# GPU压测提供测试时长和分辨率2个参数可修改，默认时长是1800秒，分辨率默认是2560x1440，请根据需求修改
# 切忌在没有理解命令的情况下，复制粘贴使用，以免犯一些低级错误
#
# 使用方法
# 拷贝文件到jetson设备的任意目录，执行以下命令：
# sudo bash benchmark.sh

SCRIPT_PATH=`pwd`

# CPU Benchmark
echo -n '[1]输入CPU核心数:[4] '
read CPU_NUM
if [ -z "$CPU_NUM" ]; then
  CPU_NUM=4
fi
gnome-terminal --title="CPU Benchmark" -- bash -c "sudo stress -c $CPU_NUM"

# GPU Benchmark
echo -n '[2]输入GPU压测时长(s):[1800] '
read GPU_DURATION
echo -n '[2-1]输入GPU压测分辨率:[2560x1440] '
read SCREEN_RESOLUTION
if [ -z "$GPU_DURATION" ]; then
  GPU_DURATION=1800
fi
if [ -z "$SCREEN_RESOLUTION" ]; then
  SCREEN_RESOLUTION='2560x1440'
fi
gnome-terminal --title="GPU Benchmark" -- bash -c "sudo glmark2 -b "terrain:duration=$GPU_DURATION" -s "$SCREEN_RESOLUTION""

# 监控开始
echo "[3]启动监控"
gnome-terminal --title="监控" -- bash -c "sudo jtop"

echo "完成脚本执行:)"
